import numpy as np

if __name__ == '__main__':
    a = [[1, 2, 3, 4, 5],
         [6, 7, 8, 9, 10]]
    print(type(a))
    b = np.array(a)
    print(type(b))
    print(b.argmax())
    print(b.mean(axis=1))

    # save
    np.save("./demo1.npy", b)
    q = np.load("./demo1.npy")
    print(q)
